Auto-sync is disabled for ready for review pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

/ok to test

Greptile Overview

Greptile Summary

This PR cleanly reverts PR #536, which removed the context parameter from memory pointer classes to improve performance. While that change successfully reduced overhead, it inadvertently broke the public API—specifically causing Apache Arrow's numba interop tests to fail with TypeError: MemoryPointer.__init__() got multiple values for argument 'pointer'.

Key Changes:

• Restores context as the first parameter to MemoryPointer, AutoFreePointer, MappedMemory, PinnedMemory, and ManagedMemory constructors
• Re-adds @require_context decorator to from_cuda_array_interface() in api.py
• Updates all instantiation sites to pass context (usually via weakref.proxy(self.context) or devices.get_context())
• Updates tests to reflect restored API

Justification:
The revert is necessary because external consumers like Apache Arrow depend on the existing signature. The performance optimization, while valuable, must be reimplemented without breaking the public API. The PR description correctly notes that targeted removal of @require_context decorators can be done in follow-up work.

Confidence Score: 5/5

• This PR is safe to merge with minimal risk—it's a clean revert that restores the previous working API
• The revert is straightforward and complete: all constructor signatures are correctly restored with context as the first parameter, all call sites are updated consistently, and tests are properly adjusted. The change fixes a critical breaking change in the public API that affected Apache Arrow integration. The diff is a nearly perfect inverse of the original PR perf: remove context threading in various pointer abstractions #536, preserving the pre-existing typo "Releease" which confirms accuracy
• No files require special attention—all changes are consistent and correct

Important Files Changed

File Analysis

Sequence Diagram

sequenceDiagram
    participant User as External User/Arrow
    participant API as from_cuda_array_interface()
    participant Context as CUDA Context
    participant MemPtr as MemoryPointer
    participant DevArray as DeviceNDArray
    
    Note over User,DevArray: Before Revert (PR #536 - BROKEN)
    User->>API: Call with cuda-array-interface
    API->>MemPtr: MemoryPointer(pointer, size, owner)
    Note over MemPtr: No context parameter!
    MemPtr-->>API: Memory pointer instance
    API->>DevArray: Create DeviceNDArray
    DevArray-->>User: Return array
    Note over User: ❌ Arrow breaks: multiple values for 'pointer'
    
    Note over User,DevArray: After Revert (Current PR - FIXED)
    User->>API: Call with cuda-array-interface
    API->>Context: current_context()
    Context-->>API: Context instance
    API->>MemPtr: MemoryPointer(context, pointer, size, owner)
    Note over MemPtr: Context restored as first param
    MemPtr->>MemPtr: Store context reference
    MemPtr-->>API: Memory pointer instance
    API->>DevArray: Create DeviceNDArray
    DevArray-->>User: Return array
    Note over User: ✅ Arrow works correctly

Following this PR: create an issue to track the work of re-removing @require_context and get_context() from from_cuda_array_interface().